#!/usr/bin/env bash
cd $(cd `dirname $0`; pwd)/..
echo $PWD
DATA_ROOT=/data/round2

rm -rf cache
mkdir cache
#cp ./resnet50_distributed_training.py ./resnet.py ./device
#cd ./device
echo "start training"
mpirun --allow-run-as-root -n 4 python ./train.py \
  --train_dir='./cache' \
  --data_url=${DATA_ROOT} \
  --device='GPU' \
  2>&1  | tee cache/train.log